import numpy as np
import matplotlib.pyplot as plt

# 设置中文显示
plt.rcParams['font.sans-serif'] = 'SimHei'
plt.rcParams['axes.unicode_minus'] = False
inputfile = './data/国民经济核算季度数据.npz'
data = np.load(inputfile, allow_pickle=True)
# 提取其中的column数组，视为数据的标签
name = data['columns']
# 提取其中的values数组，数据的存在位置
values = data['values']
print(name)
print(values)
# x轴的刻度标签
label = ['农林牧鱼', '工业', '建筑业', '批发和零售', '交通', '住宿餐饮', '金融', '房地产', '其他']
# 设置画布
plt.figure(figsize=(6, 5))
# 绘制直方图
cm = plt.bar(range(9), values[-1, 6:15], width=0.5)
# y = values[-1, 6:15]
# print(y)
# 在柱状图上方加上数值
for rect in cm:
    height = rect.get_height()
    plt.text(rect.get_x()+rect.get_width()/2.-0.2, 1.02*height, "%.2f" % height, size=10)
plt.xlabel('产业')
plt.ylabel('生产总值（亿元）')
plt.title('2020年第一季度各个行业国民生产总值柱形图（徐伟杰）')
plt.xticks(range(9), label)
plt.show()
